Staff, Software Engineer (Machine Learning Engineer) | Personalization | Real Time Inference Systems

Walmart • Washington, DC, United States • $200k - $250k / year • 2w ago

Position Summary:

Are you a seasoned Machine Learning Engineer with 5+ years of experience working on ranking, online advertising, recommender systems, ecommerce? If YES, then you are in LUCK!!

Walmart is looking to hire a highly seasoned Machine Learning Engineer for the Personalization team who would love to make an impact & change the future of retail. If you're a seasoned professional with experience in developing a Real Time Inference System, deploying large scale deep learning ML models, and implementing MLOps best practices, and if you have led design discussions for a scalable ML platform and mentored teams in an agile environment, we'd love to hear from you.

**Immigration sponsorship is not available in this role.**

About the Team: The Personalization team at Walmart is dedicated to the mission of knowing our customers better and making it frictionless and faster for them to get what they want while building a trusted partnership with Walmart. We help bring a seamless experience to our customers irrespective of how and where they engage with us – while shopping online or picking groceries at stores.

The Personalization team consists of big data geeks, application engineers, scientists and product visionaries all working together to design, prototype and build technology-driven products and experiences that will change the future landscape of e-commerce. We are focused on building an intelligent system that will enable 1:1 personalized customer experience, from finding the product to delivering it to the customer.

What you'll do:

Design and development of Real Time Inference System which will be used for deploying deep learning models or traditional ML models and run inferences on-demand at low latency.
Develop capabilities to support inferencing of Gen AI models, and support other LLM techniques like RAG, model fine tuning, building semantic cache.
Develop and implement best practices in MLOps to streamline the development, deployment, and monitoring of ML models.
Collaborate with product managers, data scientists, and other engineering teams to understand their needs and translate them into ML platform capabilities.
Lead architectural and design discussions to ensure our platform is modular, scalable, fault tolerant, and sustainably built.
Mentor and grow team members while working in an agile environment.

What you'll bring:

Bachelors degree in Computer Science, Information Technology, Engineering or related field and 7+ years' experience OR Masters degree in Computer Science, Information Technology, Engineering or related field and 5 + years' experience.
You’re experienced with building real time inference systems for deploying deep learning or traditional ML models.
Experience with developing Restful APIs using Python and Flask.
Experience with Triton Inference Server.
Experience with deploying and building real time systems for LLMs.
You have experience working on ranking, online advertising, recommender systems, ecommerce.
Working knowledge of setting up RAG.
Experience working with model lifecycle management tools like ML Flow and KF Serving.
Strong Expertise in Java, Object Oriented & Functional Programming, Design Patterns.
In-depth knowledge of No-SQL and RDBMS technologies.
Proficient with event driven architecture using Kafka.
Advanced knowledge of complex software design, distributed system design, design patterns, data structures and algorithms.
Working knowledge in container and cloud native technologies – Kubernetes, Docker.
Proficient in building scalable Microservices using Spring framework.
Experience with CI/CD tools such as Jenkins and Maven.
Strong expertise with Cloud Technologies like Azure and GCP.
Experience in monitoring production systems and using different systems like Grafana, Prometheus.
You have strong written and oral communication skills.
Experience with all phases of the software development life cycle, best practices, and Agile Software Development.

About Walmart Global Tech:
Imagine working in an environment where one line of code can make life easier for hundreds of millions of people. That’s what we do at Walmart Global Tech. We’re a team of software engineers, data scientists, cybersecurity experts and service professionals within the world’s leading retailer who make an epic impact and are at the forefront of the next retail disruption.

Flexible, Hybrid work:
We use a hybrid way of working that is primarily in office coupled with virtual when not onsite.

Benefits:
Beyond our great compensation package, you can receive incentive awards for your performance. Other great perks include 401(k) match, stock purchase plan, paid maternity and parental leave, PTO, multiple health plans, and much more.

Equal Opportunity Employer:
Walmart, Inc. is an Equal Opportunity Employer – By Choice.

Primary Location:
10500 Ne 8Th St, Bellevue, WA 98004, United States of America

#J-18808-Ljbffr